from agent.dqn import test_dqn
from agent.pg import test_pg
from agent.sac_discrete import test_discrete_sac
from agent.a2c import test_a2c
from agent.ppo import test_ppo

agent_factory = {
    'dqn': test_dqn,
    'ddqn': test_dqn,
    'dueling dqn': test_dqn,
    'pg': test_pg,
    'discrete sac': test_discrete_sac,
    'a2c': test_a2c,
    'ppo': test_ppo,
}
